﻿#region LicenseHeader

// Copyright 2012 The Trustees of Indiana University.  All rights reserved.
// 
// The Indiana University licenses this file to you under the Apache License, 
// Version 2.0 (the "License"); you may not use this file except in compliance 
// with the License.  You may obtain a copy of the License at 
// 
// http://www.apache.org/licenses/LICENSE-2.0
// 
// Unless required by applicable law or agreed to in writing, software 
// distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
//  WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.  
// See the License for the specific language governing permissions and 
// limitations under the License.
// 
// @author Thilina Gunarathne (tgunarat@indiana.edu)

#endregion

using System;
using System.Linq;
using Microsoft.WindowsAzure;
using Microsoft.WindowsAzure.StorageClient;

namespace AzureMRCore.DataModel
{
    public class MapTaskDataSource
    {
        private CloudStorageAccount storageAccount;

        public MapTaskDataSource()
        {
            storageAccount = CloudStorageAccount.FromConfigurationSetting("DataConnectionString");
        }

        public IQueryable<MapDataModel> GetMapTasks(string jobID)
        {
            CloudTableClient tableClient = storageAccount.CreateCloudTableClient();
            if (tableClient.DoesTableExist(DataModelUtils.GetMapTableName(jobID)))
            {
                MapMetaDataContext mapDataContext = new MapMetaDataContext(storageAccount.TableEndpoint.ToString(),
                                                                           storageAccount.Credentials,
                                                                           DataModelUtils.GetMapTableName(jobID));

                return mapDataContext.MapTasks;
            }
            else
            {
                throw new Exception("MapTask table for the specified job does not exist.");
            }
        }
    }
}